from .service import Service, ServiceState


class MockService(Service):
    def __init__(self, name, state):
        self._name = name
        self._state = state

    def update_state(self):
        self._state = ServiceState((self._state.value + 1) % 5 + 1)
